matplotlib animation step-by-step